# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
# Licensed under the 【火山方舟】原型应用软件自用许可协议
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at 
#     https://www.volcengine.com/docs/82379/1433703
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from arkitect.telemetry import logger
from arkitect.utils.context import get_reqid


def _attach_info():
    info = {
        "req_id": get_reqid(),
    }

    text = ""
    for k, v in info.items():
        text += f"{k}={v} "
    return text


def INFO(message: str, *args, **kwargs):
    message = f"{_attach_info()}{message}"
    logger.INFO(message, *args, **kwargs)


def DEBUG(message: str, *args, **kwargs):
    message = f"{_attach_info()}{message}"
    logger.DEBUG(message, *args, **kwargs)


def WARN(message: str, *args, **kwargs):
    message = f"{_attach_info()}{message}"
    logger.WARN(message, *args, **kwargs)


def ERROR(message: str, *args, **kwargs):
    message = f"{_attach_info()}{message}"
    logger.ERROR(message, *args, **kwargs)
